# python3 train.py --ssim --mse --model unet --epoch 100 --model_count 1 --batch-size 16 --acceleration 4
# python3 train.py --ssim --mse --model unet --epoch 100 --model_count 1 --batch-size 16 --acceleration 8


# python3 train.py --ssim --mse --model resunet --epoch 100 --model_count 1 --batch-size 16 --acceleration 4
# python3 train.py --ssim --mse --model resunet --epoch 100 --model_count 1 --batch-size 16 --acceleration 8

# python3 train.py --ssim --mse --model seresunet --epoch 100 --model_count 1 --batch-size 12 --acceleration 4
# python3 train.py --ssim --mse --model seresunet --epoch 100 --model_count 1 --batch-size 12 --acceleration 8


# python3 train.py --ssim --mse --model unet --epoch 100 --model_count 2 --batch-size 8 --acceleration 4
# python3 train.py --ssim --mse --model unet --epoch 100 --model_count 4 --batch-size 4 --acceleration 4

# python3 train.py --ssim --mse --model unet --epoch 100 --model_count 2 --batch-size 8 --acceleration 8
# python3 train.py --ssim --mse --model unet --epoch 100 --model_count 4 --batch-size 4 --acceleration 8

# python3 train.py --ssim --mse --model resunet --epoch 100 --model_count 2 --batch-size 8 --acceleration 4
python3 train.py --ssim --mse --model unet --epoch 100 --model_count 4 --batch-size 4 --acceleration 4

# python3 train.py --ssim --mse --model resunet --epoch 100 --model_count 2 --batch-size 8 --acceleration 8
# python3 train.py --ssim --mse --model resunet --epoch 100 --model_count 4 --batch-size 4 --acceleration 8

# python3 train.py --ssim --mse --model unet --epoch 100 --model_count 2 --batch-size 8 --acceleration 8
# python3 train.py --ssim --mse --model unet --epoch 100 --model_count 4 --batch-size 4 --acceleration 8
# python3 train.py --ssim --mse --model hinet --epoch 100 --model_count 1 --batch-size 16 --acceleration atio16
# python3 train.py --ssim --mse --model unet --epoch 100 --model_count 1 --batch-size 16 --acceleration 16
# python3 train.py --ssim --mse --model resunet --epoch 100 --model_count 1 --batch-size 16 --acceleration 16